package net.cyl.db;

import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
import com.baomidou.mybatisplus.generator.config.GlobalConfig;
import com.baomidou.mybatisplus.generator.config.PackageConfig;
import com.baomidou.mybatisplus.generator.config.StrategyConfig;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;

public class MyBatisPlusGenerator {
    public static void main(String[] args) {
        // 1. 全局配置
        GlobalConfig config = new GlobalConfig();
        config.setActiveRecord(false)           // 是否支持AR模式
                .setAuthor("曹玉林")           // 作者C:\Users\21407\Desktop\short-link\cloud-account\src\main\java
                .setOutputDir("/Users/21407/Desktop/short-link/cloud-account/src/main/java")  // 生成路径
                .setFileOverride(true)            // 文件覆盖
                .setIdType(IdType.AUTO)           // 主键策略
                .setDateType(DateType.ONLY_DATE)  // 日期类型
                .setServiceName("%sService")      // 设置生成的Service接口名称
                .setEntityName("%sDO")            // 实体类结尾名称
                .setBaseResultMap(true)           // 生成基本的resultMap
                .setBaseColumnList(true);         // 生成基本的SQL片段

        // 2. 数据源配置
        DataSourceConfig dsConfig = new DataSourceConfig();
        dsConfig.setDbType(DbType.MYSQL)
                .setDriverName("com.mysql.cj.jdbc.Driver")
                .setUrl("jdbc:mysql://47.109.76.105:3306/cloud_account?useUnicode=true&characterEncoding=utf-8&useSSL=false&serverTimezone=Asia/Shanghai")
                .setUsername("root")
                .setPassword("123456");

        // 3. 策略配置
        StrategyConfig stConfig = new StrategyConfig();
        stConfig.setCapitalMode(true)               // 全局大写命名
                .setNaming(NamingStrategy.underline_to_camel)  // 数据库表映射到实体的命名策略
                .setEntityLombokModel(true)         // 使用Lombok
                .setRestControllerStyle(true)       // 使用RestController注解
                .setInclude("account", "traffic", "traffic_task");  // 生成的表

        // 4. 包名策略配置
        PackageConfig pkConfig = new PackageConfig();
        pkConfig.setParent("net.cyl")
                .setMapper("mapper")
                .setService("service")
                .setController("controller")
                .setEntity("model")
                .setXml("mapper");

        // 5. 整合配置
        AutoGenerator ag = new AutoGenerator();
        ag.setGlobalConfig(config)
                .setDataSource(dsConfig)
                .setStrategy(stConfig)
                .setPackageInfo(pkConfig);

        // 6. 执行操作
        ag.execute();
        System.out.println("======= MyBatisPlus代码生成成功 ========");
    }
}